package warmachineroster.presentation.command;

import org.eclipse.gef.commands.Command;

import warmachineroster.Roster;
import warmachineroster.RosterElement;


public class RosterElementCreateCommand extends Command {
	private RosterElement rosterElement;
	private Roster roster;



	public void execute()
	{
		roster.getRosterUnits().add(rosterElement);
	}

	public void undo()
	{
		roster.getRosterUnits().remove(rosterElement);
	}

	public void setRosterElement(RosterElement rosterElement) {
		this.rosterElement = rosterElement;
	}

	public void setRoster(Roster roster) {
		this.roster = roster;
	}
}
